"""尝试使用MongoDB函数集util_mongo.py"""
from util_mongo import enqueue, get_sorted_by_key
import pymongo as pm
from common import MONGODB_NAME, VALUE, KEY, IO_PREFIX
import time

# 连接Mongodb，3秒内连不上就报错（3秒内都连不上，说明你没开mondodb，或IP、端口不正确）
# mongo = pm.MongoClient('47.95.208.117', 27017, serverSelectionTimeoutMS=3000, )
mongo = pm.MongoClient('47.95.208.117', 27017, serverSelectionTimeoutMS=3000, username='root', password='567pwd!@#')
mdb = mongo[MONGODB_NAME]

# 用户uxxxx，在当前插入聊天输入
ts = time.time_ns()  # 当前纳秒时间戳
enqueue(mdb, 'dialog_in', 'uxxxx', ts, '您好！')

# 用户uxxxx，在当前插入聊天输入
ts = time.time_ns()  # 当前纳秒时间戳
enqueue(mdb, 'dialog_in', 'uxxxx', ts, '您好！001xxxx')

# 用户uxxxx，在当前插入聊天输入
ts = time.time_ns()  # 当前纳秒时间戳
enqueue(mdb, 'dialog_in', 'uxxxx', ts, '您好！002yyyy')
print('Enqueue all OK')

# 用户uxxxx，获取最近的2条聊天输入，其他数据就不要了（清除掉）
xres = get_sorted_by_key(mdb, 'dialog_in', 'uxxxx', limit=2, is_keep_others=False)
print(xres)
print('Query all OK')
